﻿// ------------------------------------------------------------------------
// 版权信息
// 版权归重庆虫儿飞科技有限公司所有。
// 所有权利保留。
// 官方网站：https://netokit.com
// 许可证信息
// Neto.Kit 项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。
// 许可证的完整文本可以在源代码树根目录中的 LICENSE-APACHE 和 LICENSE-MIT 文件中找到。
// 
// 使用条款
// 使用本代码应遵守相关法律法规和许可证的要求。
// 
// 免责声明
// 对于因使用本代码而产生的任何直接、间接、偶然、特殊或后果性损害，我们不承担任何责任。
// 
// 其他重要信息
// Neto.Kit 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。
// 有关 Neto.Kit 项目的其他详细信息，请参阅位于源代码树根目录中的 COPYRIGHT 和 DISCLAIMER 文件。
// 
// 更多信息
// 请访问 https://netokit.com 获取更多关于 Neto.Kit 项目的许可证和版权信息。
// ------------------------------------------------------------------------

#region

using Microsoft.Extensions.Caching.Memory;

#endregion

namespace Neto.Caches;

/// <summary>
///     内存缓存
/// </summary>
public class MemoryCacheHelper
{
    /// <summary>
    ///     用于跟踪所有缓存键
    /// </summary>
    public static readonly HashSet<string> _keys = new();

    /// <summary>
    ///     获取所有缓存键
    /// </summary>
    /// <returns></returns>
    public static IEnumerable<string> GetAllKeys()
    {
        return _keys; // 返回所有缓存键
    }

    /// <summary>
    ///     设置缓存
    /// </summary>
    /// <param name="cacheKey">缓存key</param>
    /// <param name="value">值</param>
    /// <param name="options">缓存选项</param>
    public static void Set(string cacheKey, object value, MemoryCacheEntryOptions? options = null)
    {
        value.SetMemoryCache(cacheKey, options);
    }


    /// <summary>
    ///     设置缓存
    /// </summary>
    /// <param name="cacheKey">缓存key</param>
    /// <param name="value">值</param>
    /// <param name="options">缓存选项</param>
    public static async Task SetAsync(string cacheKey, object value,
        MemoryCacheEntryOptions? options = null)
    {
        await value.SetMemoryCacheAsync(cacheKey, options);
    }

    /// <summary>
    ///     获取缓存数据
    /// </summary>
    /// <typeparam name="TEntity"></typeparam>
    /// <param name="cacheKey">缓存Key</param>
    /// <param name="func">方法</param>
    /// <param name="options">缓存配置选项</param>
    /// <returns>返回的实体对象</returns>
    public static TEntity? Get<TEntity>(string cacheKey, Func<TEntity>? func = null,
        MemoryCacheEntryOptions? options = null)
    {
        return cacheKey.GetMemoryCache(func, options);
    }

    /// <summary>
    ///     获取缓存数据
    /// </summary>
    /// <typeparam name="TEntity"></typeparam>
    /// <param name="cacheKey">缓存Key</param>
    /// <param name="func">方法</param>
    /// <param name="options">缓存配置选项</param>
    /// <param name="token">取消令牌</param>
    /// <returns>返回的实体对象</returns>
    public static async Task<TEntity?> GetMemoryCacheAsync<TEntity>(string cacheKey,
        Func<Task<TEntity>>? func = null,
        MemoryCacheEntryOptions? options = null, CancellationToken token = default)
    {
        return await cacheKey.GetMemoryCacheAsync(func, options, token);
    }


    /// <summary>
    ///     获取默认内存缓存选项
    /// </summary>
    /// <param name="options"></param>
    /// <returns></returns>
    public static MemoryCacheEntryOptions GetDefaultOptions(MemoryCacheEntryOptions? options = null)
    {
        options ??= new MemoryCacheEntryOptions();
        //{
        //SlidingExpiration = new System.TimeSpan(1, 1, 1, 1),
        //AbsoluteExpiration=DateTimeOffset.Now.AddDays(10)
        //};
        options.RegisterPostEvictionCallback();
        return options;
    }


    /// <summary>
    ///     移除内存缓存
    /// </summary>
    /// <param name="cacheKey"></param>
    public static void Remove(string cacheKey)
    {
        cacheKey.RemoveMemoryCache();
    }

    /// <summary>
    ///     移除内存缓存
    /// </summary>
    /// <param name="cacheKey"></param>
    /// <returns></returns>
    public static async Task RemoveAsync(string cacheKey)
    {
        await cacheKey.RemoveMemoryCacheAsync();
    }

    /// <summary>
    ///     是否存在缓存Key
    /// </summary>
    /// <param name="cacheKey">缓存Key</param>
    /// <returns></returns>
    public static bool Exist(string cacheKey)
    {
        return cacheKey.ExistMemoryCache();
    }

    /// <summary>
    ///     是否存在缓存Key
    /// </summary>
    /// <param name="cacheKey">缓存Key</param>
    /// <returns></returns>
    public static async Task<bool> ExistAsync(string cacheKey)
    {
        return await cacheKey.ExistMemoryCacheAsync();
    }
}